import getpass
db = {}

def zhuce():
    '用于用户注册'
    user = input('请输入用户名：').strip()
    if user == '':
        print('用户名不能为空')
    elif user in db:
        print('用户已存在')
    else:
        passwd = input('输入密码：')
        db[user] = passwd
        print('\033[32;1m已自动登入\33[0m')

def login():
    '用于用户登陆'
    user = input('用户名：').strip()
    passwd = getpass.getpass('输入密码')
    if db.get(user) == passwd:
        print('登入成功')
    else:
        print('登入失败')

def show_menu():
    '用于显示程序的主菜单，根据用户的选择调用相关的函数'
    funcs = {'0':zhuce,'1':login}
    prompt = '''
    (0) 注册
    (1) 登陆 
    (2) 退出
    ('请输入0/1/2：')'''
    while 1:
        xuanze  = input(prompt).strip()
        if xuanze not in ['0', '1', '2']:
            print('无效的输出，请重试')
            continue
        if xuanze == '2':
            print('ByeBye')
            break
        funcs[xuanze]()

if __name__ == '__main__':
    show_menu()